Improved Graph Convolutional Neural Network for Dance Tracking and Pose Estimation

Movement recognition technology is widely used in various practical application scenarios, but there are few researches on dance movement recognition at present. Aiming at the problem of low accuracy of dance movement recognition due to complex pose changes in dance movements, this paper designed an improved graph convolutional neural network algorithm for dance tracking and pose estimation. In this method, the spatial and temporal characteristics of motion are extracted from the skeleton joint diagram of human body. Then, GCN (graph convolutional neural) is used to extract potential spatial information between skeleton nodes. Finally, LSTM (long short-term memory) extracts the time series features before and after human actions as a supplement and performs late fusion of the prediction outputs of the two networks, respectively, to improve the problem of insufficient generalization ability of single network. The experimental results show that this method can effectively improve the accuracy of dance movement recognition in general movement recognition data set and dance pose data set. It has certain application value in dance self-help teaching, professional dancer movement correction, and other application scenarios.


Introduction
In the field of computer vision, human motion recognition is a hot research topic in recent years [1]. With the continuous exploration and research of scholars in the past decade, many excellent achievements have been made in human motion recognition, and this technology has been applied to all walks of life [2]. Each dance usually has some symbolic movements, gestures, props, etc., which can be used as key elements to record and classify each dance and can be used to build a database of a few dance movements and provide data support for more innovative applications [3].
In the field of dance research, some technical schemes for digital preservation and display of dance have emerged, but dance movements are almost recorded by large 3D motion capture equipment [4]. ese devices are expensive and have poor flexibility, which affects the execution and recognition of actions. In addition, for the problem of occlusion in human movements, the motion characteristics in different scenes have still not achieved satisfactory effects, which also limits the inheritance and protection of dance [5]. For motion data collection, Kinect depth vision sensor equipment has the advantages of high resolution of depth map, low cost, and ability to directly track the motion trajectory of human skeleton [4]. Moreover, using skeleton information for motion recognition has two advantages. First, bone data is an abstraction of human body information in three-dimensional space, and it has certain robustness to noise such as background and light, thus providing a good representation of human behavior. Second, compared with RGB video data, bone data has a smaller data dimension, which also makes it possible to design lightweight and hardware friendly models.
With the deep integration of technology and culture, the application of computer vision technology in dance teaching has great potential [6]. General subjects follow the teaching mode of teachers speaking and students listening, but dance is performed by body, so teachers need to judge the standard degree of learners' movements and use different teaching methods and means. In traditional dance teaching, it is a major problem for learners to judge whether movements are standard or not [7]. Learners can only rely on their own subjective judgment and teacher's evaluation to judge whether their actions are standard or not. However, the traditional dance class has a large number of students and belongs to one-to-many teaching mode, so the teacher cannot give real-time guidance to every learner. In traditional dance teaching, although there is a unified standard of dance movements, the judgment of the standard of learners' dance movements mainly relies on subjective evaluation, lacking a quantifiable objective evaluation method. Applying information technology and exploring new teaching methods will provide infinite possibilities for the reform of dance teaching.
With the development of deep neural network, many human pose estimation methods based on deep neural network have been proposed in recent years. Top-down human posture estimation methods include CPM [8], Hourglass [9], Simple Baselines [10], and HRNet [11]. And bottom-up human pose estimation methods include OpenPose [12], DeepCut [13], and HigherHRNet [14]. However, there are few studies on motion similarity estimation, and most of them focus on motion recognition. For example, Zemike moment is used to describe the shape information of images in [15], and Support Vector Machine (SVM) is used to identify the movements of people. Literature [16] uses PAFs method to identify key information of human body and uses LSTM network to classify the information, so as to achieve the purpose of identifying human actions. In [17], the image is preprocessed and then modelled and analyzed to identify the actions of the characters. ese methods only recognize the actions of the characters but do not judge whether the actions are standard or not and cannot provide suggestions for the improvement of the gestures for learners. Based on the above problems, this paper proposes a dual-flow network method based on the combination of GCN and LSTM, 2S-LSGCN. is paper has three main contributions as follows.
(1) GCN network is mainly used to extract spatial feature information implied in human skeleton. (2) Bi-LSTM was used to extract the time feature information of the actions in the complete skeleton graph as a supplement, and a time feature subsampling layer was added before input into the LSTM network to extract rich and abstract time sequence features. (3) e predicted outputs of the dual-flow network are, respectively, late fused to obtain the final predicted output value of the dual-flow network.
is paper consists of five main parts: the first part is the introduction, the second part is state of the art, the third part is methodology, the fourth part is result analysis and discussion, and the fifth part is the conclusion.

State of the Art
A spatial-temporal Graph Convolutional neural Network (ST-GCN) based on bone data can extract spatial-temporal features from both time and space. e main process is as follows. Firstly, the graph structure data representing the bone sequence of a given action video is constructed and used as the input of ST-GCN. en, a series of spatiotemporal convolution operations are performed to extract highlevel spatiotemporal features. Finally, Softmax classifier is used to obtain the classification results. ST-GCN realizes end-to-end training, and its overall structure and convolution unit structure are shown in Figures 1 and 2.
Each graph convolution unit contains a spatial convolution layer, a temporal convolution layer, and a residual structure. ere is a learnable edge weight parameter in the convolution unit of the space-time graph, which is used to learn the importance of edges between nodes. e advantage of this edge weight is that it can be used as an internal attention mechanism of ST-GCN to strengthen relevant information. However, the deficiency is that the key nodes and structural features are not effectively strengthened. ST-GCN time convolution layer adopts convolution operation with fixed structure, and the convolution kernel size is single definite value. Firstly, the spatial structure features are obtained through the spatial convolution layer, and then the time features in the direction of time flow are obtained through time convolution. Finally, the residual mechanism is used to fuse the original input and the spatiotemporal features of the high-level to form the output of each spatiotemporal convolution unit. In each ST-GCN space-time convolution unit, spatial graph convolution and temporal graph convolution can have similar expressions.
I out q nx � q n ∈ H q n 1 K nx q ny I in q ny · m L nx q ny , H q nx � q vy d q ny , q nx ≤ Z, v − n w l ≤ R 2 . (1) 1/K nx (q ny ) is the normalized term, q nx represents the target node, and q ny represents the neighbor node of nx. M() is the weight function. H(q nx ) is the sampling function. L nx represents the label graph with q nx as the target center. R represents the size of time convolution kernel. d(q ny , q nx ) is the shortest path from node q nx to q ny . Z is the set policy parameter. q vy represents the set of sampling nodes, v stands for sample frame, and n w stands for target frame. e process of obtaining spatiotemporal correlation information through spatiotemporal graph convolution is shown in Figure 3. e input bone data contains two sequences: spatial sequence and time sequence. e spatial sequence is the sequence of key nodes in a single frame, and the time sequence is the sequence of time stream. Firstly, the structural features of each single frame are extracted from the spatial sequence through the spatial convolution layer. ese features not only contain node features and the association information between nodes within the frame, but also retain the original time series. en, interframe features are extracted from the direction of time stream through the time convolution layer. Interframe features include the track features of nodes, namely, time association information.
rough the above two steps, spatiotemporal convolution obtains not only the structured spatial correlation information in a single frame, but also the temporal correlation information of node trajectories between consecutive frames. us, a feature map containing a large amount of spatiotemporal correlation information is formed.

Acquisition of Human Skeleton Features.
In the task of dance motion tracking and pose estimation of video stream, most researchers only focus on the pixel information in RGB video stream. It ignores the performance of human movements and is mainly completed by mutual traction and cooperation between skeleton and joints. erefore, human skeleton joint diagram contains abundant information of action characteristics. However, most of the action recognition data sets, such as HMDB, 20BN-Jester, Kinetics, etc., have only RGB video or image samples and do not have the human point information.
At present, there are mainly two methods to obtain the characteristic information of human body joints in the sequence stream: ① To capture the depth information of the movement of people in the three-dimensional space through Kinect (3D motion camera) and then to obtain the coordinates of bone points from the depth image to form the joint diagram of human skeleton. ② e 2D pose estimation algorithm (such as Open) can be used for the RGB motion

Residual mechanism
Spatial graph convolution Time domain graph convolution  Computational Intelligence and Neuroscience video stream (Pseo) to extract the 2D coordinates of the joints and bone information between the joints.

Graph Convolutional Neural Network.
Since the human skeleton spatiotemporal graph input into the network is irregular non-European spatial data, the traditional convolutional network (CNN) cannot be directly applied to extract the features of the graph data, so graph convolution is used to extract the local location features of the node space. For the spatial dimension of skeleton space-time graph, the convolution operation of graph convolution operation for each vertex q x is as follows: where f out represents the feature structure, q y represents the vertex in the graph, and H x represents the receptive field of the convolution operation on q x (the distance between the convolution center and its neighbor node is defined as 1). M is the weight function, similar to the traditional convolution (CNN).

LSTM Network.
RNN has become a common method in video sequential tasks, and LSTM is the best RNN at present. Mainly due to its long time memory, when processing the sequence, the output features of this moment will be input together with the sample information of the next moment. In this way, the time information of the sequence is well preserved and the perceptual ability of the model to the information between the action frames is greatly improved. LSTM is usually used to deal with time-dependent sequence problems, but LSTM's memory capacity is also limited, while Bi-LSTM has a stronger ability to deal with time series. Bi-LSTM uses two layers of LSTM connected in different directions to capture the deep spatiotemporal characteristics of three-dimensional bone coordinates evolving over time.
e LSTM network consists of three gates (input gate x n , forget gate f n , and output gate o n ), unit state c n , and hidden state b n . Among them, LSTM's ability to extract correlation information from time series mainly benefits from the clever design of unit state c n and hidden state b n , which enables LSTM to choose to discard or retain features with temporal significance. e specific calculation is as follows: 2S-LSGCN is a combination of LSTM and GCN. e graph convolutional neural network (GCN) is used to extract the spatial relationship features of nodes in the skeleton graph, and the bidirectional long and short memory network (Bi-LSTM) with stronger time memory ability is used to extract the time feature information of action sequences in the skeleton graph. Specifically, the idea of the model is to use 2D attitude estimation algorithm to calculate the 2-dimensional coordinates (i, j, c) of the human body's key points in each frame space from the action video stream, where i and j, respectively, represent two-dimensional coordinates and c represents confidence degree. e skeleton joint graph composed of joint features in time and space was input into GCN network and LSTM network, respectively. As the time series of the original skeleton joints are too long and the duration of each video sample is different, LSTM network cannot effectively extract the feature relationship before and after the time series. e time dimension in the skeleton joint diagram was reduced by adding subsampled layers. Finally, the prediction output of dual-stream network is combined with late fusion to solve the problem of insufficient generalization ability of single network. e model in this paper is shown below (see Figure 4).

Skeleton Joint Diagram Construction.
In this paper, the graph structure data-skeleton joint diagram is used as the input of LSTM and GCN, respectively. e following describes the construction method of skeleton joint diagram. Firstly, 2D pose estimation algorithm is used in this paper. Open Pseo is used to obtain the coordinate information of each frame node in the video, and the coordinates of different positions of each frame node are combined into the skeleton joint space-time map.
Specifically, all videos need to be adjusted to a smaller resolution (340 × 256) and converted to a 30FPS frame rate. e purpose is to improve the accuracy of attitude estimation and reduce the reasoning time of the model. Second, the Open Pseo toolbox was used to estimate the positions of the 18 joints in each frame. e toolbox provides 2D coordinates (i, j) in pixel coordinates and gives confidence scores c for 18 human joints. erefore, we use the (i, j, c) tuple to represent the two-dimensional coordinate information of each joint. e 2D pose estimation algorithm only obtains the coordinate information of the joints, but the spatial and temporal dimensions between the joints are not connected and cannot be directly input into the graph convolutional neural network. erefore, the complete skeleton joint map should be established according to the joint coordinates first. Figure 5 shows the constructed spatiotemporal skeleton joint diagram, where the joints are represented as vertices and their natural connections in the human body are represented as edges.

Space GCN Network.
Compared with action RGB video stream action, skeleton joint diagram has smaller feature dimension. erefore, the network based on graph structure as input has not only doubled its running speed, but also far less computation than traditional CNN network. However, it also brings a difficult problem, that is, how to maintain high recognition accuracy under the condition of simple spatial topology and lack of time series information, that is, how to extract abstract high-dimensional features from a small number of low-dimensional features and then accurately predict the categories of actions. Starting from this problem, this paper uses multilayer stacked deep neural network, specifically stacking multiple graph convolution layers with different input and output dimensions and then extracting rich high-dimensional features. Secondly, the number of human nodes is generally less than 30, so the convolution kernel with a small scale of 3 * 3 is used to greatly reduce the receptive field, and the joint information with strong spatial dependence can be extracted more centrally. Finally, the experimental results show that the prediction results of graph convolutional networks with less than 9 layers will be greatly reduced. When it is larger than 9 layers, the accuracy of the network does not improve, but the amount of computation increases exponentially. erefore, the model determines the number of graph convolution layers at 9 layers. ere is no fixed number of neighbor nodes for each vertex in the skeleton space-time graph, so the convolution operation in the graph structure data needs to define the mapping function l x so that each vertex corresponds to a unique weight vector. According to the results in ST-GCN, the optimal mapping function is obtained by defining the segmentation strategy according to the distance from the center of gravity. e specific calculation is as follows: l x q y � S 1 ifr y � r x , S 2 ifr y < r x , S 3 ifr y > r x .
e strategy sets the size of the graph convolution kernel as 3, and the receptive field H x is divided into three subsets:

Computational Intelligence and Neuroscience
(1) the root node of S 1 itself; (2) S 2 centripetal subset, that is, the neighbor node is closer to the center of gravity; (3) the S 3 centrifugal subset, that is, the neighbor node is farther from the center of gravity.
With the deepening of network depth, the gradient becomes smaller and smaller after multiple convolution and multiplication, and the problem of gradient dissipation appears. Moreover, the difference between the input dimension and the output dimension is too large, and the extracted feature is too abstract, which is far from the original feature information, leading to the lower accuracy of motion prediction. erefore, skip Connect is introduced in this paper to fuse the input characteristics of the convolution layer with the output of this layer, and the calculation mode is shown in the formula below.
where i 1 and i l+1 represent the input and output features of graph convolution layer, respectively, F represents a series of nonlinear transformations in this layer, and M l is the set of weight parameters. If the characteristic dimension of the output changes after the feature passes through the convolution layer, upsampling or downsampling should be added to the residual formula, depending on the specific situation. e modified formula is as follows: in order to avoid overfitting in this layer and reduce the number of network parameters. e output features of graph convolutional network are globally pooled, and the final loss function is ReLu function. e input is nonlinear processing, and the prediction results are output.
3.6. Time LSTM Network. Short and long memory network is a recurrent neural network (RNN), which can remember the feature relationships of a long time series. Since graph convolutional neural network is only suitable for simple time convolution kernel, the time dimension of skeleton joint graph is processed. erefore, only the features that change before and after some nodes are extracted, but the rich time information of video stream is lost. erefore, bi-long and short memory network (Bi-LSTM) is used as a supplement for the model in this paper. Bi-LSTM can learn sequential and reverse time information at the same time, so as to enhance the model's ability to extract sequential information.
Different from the end-to-end CNN-LSTM network, the high-dimensional features extracted by CNN are input into LSTM. In the model presented in this paper, the Bi-LSTM network uses the original unprocessed skeleton joint spacetime map as input, so it retains richer original temporal features. Specifically, BL-1 and BL-2 represent the first and second Bi-LSTM layers. Stack BL-1 and BL-2 layers together, and the output of BL-1 serves as the input of BL-2. It is expressed by the following formula: where f out is the output of Bi-LSTM at the second layer, f HL−1 ( * ) is the feature extraction function of Bi-LSTM at the first layer, and f HL−2 ( * ) is the feature extraction function of Bi-LSTM at the second layer. M 1 and M 2 are the weight parameters of BL-1 and BL-2 layers, respectively. Due to the length of time feature sequence in input original skeleton joint space-time diagram is too long, and the duration of each video sample is different. Take the running video V1 of 10 s as an example; extract an input sample representing the video i 1 , i 2 , i 3 , . . . , i n−1 , i n , where n � 300, namely, the input time characteristic dimension n � 300. If the sample data is directly input into the Bi-LSTM network, the input dimension in the network must also be equal to the sample feature dimension. However, the memory capacity of the recurrent neural network is limited, and Bi-LSTM cannot learn the pre-and postcorrelation from such a long time feature as n � 300 and will greatly increase the weight number of the network and consume a lot of computing resources.
Based on the above problems, before the skeleton joint diagram is directly input into Bi-LSTM network, the subsampled layer is introduced in this paper. By downsampling the time dimension of samples, a time series with shorter time dimension and more abstract features is obtained. It is found that mean pooling and 1 * 1 convolution are two common and effective downsampling strategies. Mean pooling has excellent effect on dimensionality reduction of image features. However, the biggest difference between graph structure data and Euclidean spatial data is that there are spatial topological relationships among vertices in the graph, and mean pooling will lose such important topological connections. erefore, the 1 * 1 convolution operation is adopted in this paper to conduct downsampling dimension reduction for the time dimension of skeleton joint graph.

Dual-Stream 2S-LSGCN Network.
e GCN is fused with the improved Bi-LSTM network to form 2S-LSGCN dual-current network, which not only greatly improves the accuracy of recognition, but also is a powerful feature extraction network running in parallel and synchronism. GCN is used to extract spatial information between input skeleton nodes, and the improved Bi-LSTM is used to extract timedependent temporal features, so as to supplement the deficiency of GCN in time sensitivity. e late fusion of the prediction results of the two networks is carried out to obtain the final forecast output value of the dual-stream network. Specifically, the final detection performance is improved by combining the detection results of GCN and Bi-LSTM networks through the Add Connection strategy. e specific formula is as follows:

Experimental
Steps and Settings. In this paper, the OpenPose method is firstly used to extract bone joint coordinates from the video, and the missing data and illegal values are processed through data cleaning. In order to enhance the generalization ability, the joint coordinates are normalized, simulated camera movement, data filling, and other operations to enhance the data. On this basis, the edge weight matrix of each video is calculated, and the edge weight matrix and bone data are input into the network for training.

Experimental Evaluation Based on Kinetics Data Sets.
Using Kinetics data set 1, the proposed method was compared with five other motion recognition methods [18][19][20][21][22]. Kinetics' data training set consisted of 15,440 videos and test set consisted of 1,241 videos, each with a bone sequence length of 300 frames. 65 epochs are trained by default for each network. e random gradient descent of 0.1 learning rate is used for learning, and the learning rate of every 10 epochs is reduced by 0.01. e dimension E of the delay vector is set to 4 and the step τ is 1 when calculating the edge weight matrix. In this paper, top-1 and top-5 accuracy rates were used to evaluate the effectiveness of the experimental effect. e higher the accuracy rate, the better the recognition effect.
As can be seen from Table 1 and Figure 6, due to the addition of edge weight causal correlation, the accuracy of actions in this paper is higher than other methods. Compared with the method in [18], top-1 and top-5 improved by 9% and 10.94%, respectively. Compared with [19], top-1 and top-5 improved by 4.44% and 4.32%, respectively. Compared with [20], top-1 was improved by 3.81% and top-5 by 2.54%. Compared with [21], top-1 and top-5 improved by 3.29% and 3.33%, respectively. Compared with [22], top-1 and top-5 improved by 2.6% and 2.41%, respectively.
As listed in Table 2, compared with the graph convolutional network model, the top-1 accuracy of the proposed method in some action classification is improved by 3% to 22%. Although recognition is slightly lower for certain actions such as kicking the hind and small jump combination. is is due to the uneven picture quality of Kinetics data sets, the low degree of specification of movements, and the fact that many movement categories are not strongly associated with physical movements. Since the graph   Computational Intelligence and Neuroscience

Experimental Evaluation Based on Dance Data Sets.
is paper also constructs a real modern dance video data set. e data set, collected from 132 contemporary dance students, contains 2,956 video clips subdivided into 25 movements. Figure 7 shows an example of a modern dance data set.
As shown in Table 3 and Figure 8, most models can achieve good recognition results in modern dance data sets. e proposed method can achieve the performance for the following reasons: (1) e modern dance videos collected in this paper are all shot in front, and the subjects are always kept in the picture. (2) ere are almost no extra objects blocking the human body and background interference in the video scene, which makes the extracted bone data have high integrity and accuracy. Integrity is very important for motion recognition. (3) Subjects perform modern dance routines with a high degree of completion and less nonstandard movements, which enables the network to better extract features. Due to the addition of causal coefficient as edge weight, this paper can highlight the main joints in the process of human movement, and its effect is still better than other methods. is shows that the graph convolution network based on joint causality is more biased towards some nodes. e model has been applied to the collection of modern dance movement library.

Convergence
Performance Test. Figure 9 shows the comparison diagram of loss convergence between the other four models and the improved graph convolutional neural network model designed in this paper under the same parameter setting and training setting. It shows that, under the same training rounds, the convergence speed of multiscale spatiotemporal graph convolution model is faster than other models, and the effect is better.

Conclusion
Traditional dance teaching and training can only appraise whether its movements are standard through vision. It lacks objective quantified evaluation index problem. Combined with the practical application requirements of dance training, this paper proposes a dance tracking and pose estimation algorithm based on improved graph convolutional neural network, that is, a dual-flow network method based on the combination of GCN and LSTM. Unlike the traditional RGB image as input to the network, 2S-LSGCN uses as input a skeleton joint map composed of human joint coordinates. en, GCN was used as a spatial feature extractor, and Bi-LSTM was used to extract temporal inverse information. A 1 * 1 convolution subsampling layer is added to the sequential flow network to extract rich and abstract time features. Finally, the final forecast output value of dual traffic network is obtained by late fusion of the predicted output value of dual traffic network. Experimental evaluations on Kinetics public data sets and constructed real modern dance data sets demonstrate that the proposed method can effectively learn gesture features and improve the accuracy of dance movement recognition. In the following research, the problem of movement recognition for dancers with complex dance postures will be studied in more depth, Training samples Proposed Literature [21] Literature [23] Literature [20] Literature [22] Literature [19]  Computational Intelligence and Neuroscience and an intelligent dance-assisted training system with better performance is expected to be built.

Data Availability
e labeled data set used to support the findings of this study is available from the corresponding author upon request.

Conflicts of Interest
e author declares no conflicts of interest.